//package com.product.system.cwebsocket;
//
//import cn.hutool.json.JSONObject;
//import cn.hutool.log.Log;
//import cn.hutool.log.LogFactory;
//import com.alibaba.excel.util.StringUtils;
//import com.alibaba.fastjson.JSON;
//import com.product.system.service.SysRoleService;
//import org.springframework.beans.factory.annotation.Autowired;
//import org.springframework.stereotype.Component;
//
//import javax.websocket.*;
//import javax.websocket.server.ServerEndpoint;
//import java.io.IOException;
//import java.util.HashMap;
//import java.util.Map;
//import java.util.concurrent.ConcurrentHashMap;
//
//@ServerEndpoint("/websocket")
//@Component
//public class WebSocketServer {
//    static Log log = LogFactory.get(WebSocketServer.class);
//    /**静态变量，用来记录当前在线连接数。应该把它设计成线程安全的。*/
//    private static int onlineCount = 0;
//    /**concurrent 包的线程安全 Set，用来存放每个客户端对应的 MyWebSocket 对象。*/
//    private static ConcurrentHashMap<Session, WebSocketServer> webSocketMap = new ConcurrentHashMap<>();
//    /**与某个客户端的连接会话，需要通过它来给客户端发送数据*/
//    private Session session;
//    @Autowired
//    private SysRoleService sysRoleService;
//
//    /**
//     * 连接建立成功调用的方法
//     */
//    @OnOpen
//    public void onOpen(Session session) {
//        this.session = session;
//        webSocketMap.put(session, this);
//        addOnlineCount();
//        log.info("用户连接，当前在线人数为:" + getOnlineCount());
//
//        try {
//            Map<String, Object> map = new HashMap<>();
//            map.put("server", sysRoleService.showForm());
//            map.put("pie", sysRoleService.showForm());
//            JSONObject jsonObject = new JSONObject(map);
//            sendMessage(jsonObject.toString());
//        } catch (Exception e) {
//            log.error("网络异常!!!!!!");
//        }
//    }
//
//    /**
//     * 连接关闭调用的方法
//     */
//    @OnClose
//    public void onClose() {
//        if (webSocketMap.containsKey(session)) {
//            webSocketMap.remove(session);
//            subOnlineCount();
//        }
//        log.info("用户退出，当前在线人数为:" + getOnlineCount());
//    }
//
//    /**
//     * 收到客户端消息后调用的方法
//     *
//     * @param message 客户端发送过来的消息
//     */
//    @OnMessage
//    public void onMessage(String message, Session session) {
//        log.info("用户消息，报文:" + message);
//        // 可以群发消息
//        // 消息保存到数据库、redis
//        if (StringUtils.isNotBlank(message)) {
//            try {
//                // 解析发送的报文
//                com.alibaba.fastjson.JSONObject jsonObject = JSON.parseObject(message);
//            } catch (Exception e) {
//                e.printStackTrace();
//            }
//        }
//    }
//
//    /**
//     *
//     * @param session
//     * @param error
//     */
//    @OnError
//    public void onError(Session session, Throwable error) {
//        log.error("用户错误，原因:" + error.getMessage());
//        error.printStackTrace();
//    }
//
//    /**
//     * 实现服务器主动推送
//     */
//    public void sendMessage(String message) throws IOException {
//        this.session.getBasicRemote().sendText(message);
//    }
//
//    public static void sendAllMessage(String message) throws IOException {
//        for (Session session : webSocketMap.keySet()) {
//            WebSocketServer webSocketServer = webSocketMap.get(session);
//            webSocketServer.session.getBasicRemote().sendText(message);
//            System.out.println("webSocket 实现服务器主动推送成功");
//        }
//    }
//
//
//
//    public static synchronized int getOnlineCount() {
//        return onlineCount;
//    }
//
//    public static synchronized void addOnlineCount() {
//        WebSocketServer.onlineCount++;
//    }
//
//    public static synchronized void subOnlineCount() {
//        WebSocketServer.onlineCount--;
//    }
//}